UniPinRec: Unifying Generative Retrieval and Ranking at Pinterest Scale
UniPinRec:在 Pinterest 规模统一生成式检索与排序
🧭 统一检索与排序的全栈推荐 ★ 深度解读
通过掩码动作建模、联合训练与跨进程 KV 缓存共享,实现检索和排序的全栈统一,线上提升互动并降低延迟。
📌 工作参考价值(4 点)
- 掩码动作建模(MAM):将排序监督直接融入检索的序列输入,随机掩码历史动作避免序列长度膨胀,实现检索与排序共享同一主干,可快速复用到多任务推荐场景。
- 跨阶段 KV 缓存共享:通过 GPU 内存池 + CUDA IPC 实现进程间零拷贝传输,排序阶段复用检索生成的用户历史 KV,将排序的计算量从 O(n²) 降至 O(nk),显著降低在线延迟。
- 统一训练数据构造:采用 feedview(展示集)作为排序监督,构建“过去行为序列 + 未来展示列表”的统一样本,结合 Ray 在线桶连接避免数据膨胀,为级联模型的联合训练提供了可落地的数据管线范式。
- 增量上线策略:先替换检索+轻量排序(L0+L1),再逐步扩展到精排,保持与现有系统兼容,支持分阶段 A/B 测试和回滚,降低统一模型部署风险。